#
import unittest
from apps.irhw.lfm_dss import LfmDss

class TLfmDss(unittest.TestCase):
    def test_load_ds(self):
        dss = LfmDss(ds_fn='irhw_dss/test.txt')

    def test_filter_ds(self):
        src_ds = 'D:\\research\\pykan\\work\\datasets\\irhw_dss\\train.txt'
        dst_ds = 'D:\\research\\pykan\\work\\datasets\\irhw_dss\\test.txt'
        LfmDss.filter_ds(src_ds=src_ds, dst_ds=dst_ds)